Repository navigation
[Prim] add reduce_as op for paddle - #63064
Merged
Merged
Conversation
|
你的PR提交成功,感谢你对开源项目的贡献! |
|
❌ The PR is not created using PR's template. You can refer to this Demo. |
| self.python_api = paddle.sum_as | ||
| self.public_python_api = paddle.sum_as | ||
| self.op_type = "sum_as" | ||
| self.prim_op_type = "prim" |
Collaborator
There was a problem hiding this comment.
这个算子目前没有反向拆解,这里是不能设置的
Collaborator
|
补充动态shape的单侧 |
Contributor
Author
|
由于在本地cuda 11.2环境, 本地cuda12.0环境 以及CI环境都无法复现test_assign_pos_op 单测出错的场景,现将该PR的实现部分进行拆解,分成多个子PR,确认是哪个部分的改动引起了这个单测出错。这些PR并不需要Review和Merge具体PR如下: reduce_as 前向计算: #63652 |
cyber-pioneer
approved these changes
Apr 19, 2024
zyfncg
approved these changes
Apr 19, 2024
jeff41404
reviewed
Apr 19, 2024
| 'float64', | ||
| 'int16', | ||
| 'int32', | ||
| 'int64', |
Contributor
There was a problem hiding this comment.
can we support other data type of complex64/128、uint8 andint8?
luotao1
approved these changes
Apr 19, 2024
30 tasks done
Galaxy1458
added a commit
that referenced
this pull request
Apr 22, 2024
This reverts commit 38182e7.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
PR Category
Others
PR Types
New features
Description
由于
broadcast类算子的反向拆解无法适配动态shape,在静态shape中的例子如下,
但在动态shape的编译期拿不到
reduce_dim,于是要添加reduce_as算子,其伪代码如下:有了这个算子之后,动态shape的场景写法如下:
这个pr就是实现对
reduce_asop的添加